# 3 - 2 - 2 阶跃函数的实现
import numpy as np
import matplotlib.pylab as plt

def step_func(x):
    if x > 0:
        return 1
    else:
        return 0
    
def step_function(x):
    return np.array(x > 0, dtype=np.int32)

if __name__ == "__main__":
    x = np.arange(-5.0, 5.0, 0.1)
    y = step_function(x)
    plt.plot(x, y)
    plt.ylim(-0.1, 1.1)
    plt.show()
